Simple python code for a simulation class.