Got momentum sort of working