write structure for simulator
add colours, pygame import, simulated data import