Adding momentum WIP