Added momentum, (still doesn't work though)