Added gravity control and added momentum