Fixed momentum bug - annealing no longer needed