Actually integrating the momentum into the new code
clean up