moved to centre of mass frame and added more comments