Add colouring based on ScheduleState