Adding the main simulation class