SimulationScreen class enhanced