Refactoring the code to handle momenta correctly (the previous code was not general enough for all the options i've ended up using with HMC).