4c77bae772
* add momentum * make tests pass for momentum * blacking * lint Co-authored-by: Andrew Zhao Luo <andrewzhaoluo@system76-pc.localdomain>