Add args

Make the coord that shall be shifted configurable. It's not always called "x".
Check what happens if there's not halos. Just setting to 0 won't do, because it's not clear how the then empty new_halo would be have. Better use explicit if number_of_halo_points == 0: ....