Momentum function worked on