Time stepping tests

Tested some simple time stepping. Mainly for myself to learn how
classes and functions in python.