Added momentum attribute, tested and updated all tests
Cleaned up the tests a little, adding menu and sorting console