Started writing simulation code

Need to clean up interface for plotting