1. Yes, its on the aux belt, so if its been taken off and you still have the noise it is not that.
2. The balance shaft module is at the bottom of the engine block, inside the sump. When they replaced the oil pump and head they would not have to change the balance shaft assembly also. They should have, but chances are they did not as it is not a serviceable part, however it is a part that is known for failing!
You can drop the oil, put a small inspection camera up through the oil drain plug and you will see the balance shaft module. If you see a chain at the front of it (towards front of the car) it has not been changed and chances are that is what you are hearing, rattly chain with a broken tensioner which will catastrophically fail in the near future, potentially killing the whole engine with it.
If you see a gear driven mechanism on the balance shaft, then it has been changed and what you are hearing is not the rattly chain. All you need is a £10 eBay USB inspection camera with LED's on the front to light up inside the engine, and then raise the car, drop the oil and have a look.