Changing the simulation environment to include cubes
Same as before