Fixed a small bug and added print statement in main
clean points done with tests