Make momentum a comonad