Finished timestepping kernels and python code