Split Update/Step via subclassing in methods/

[matt-chan]

Time: 3

• Use subclasses to avoid if/else trees.
• docs
• unit tests