Fixing bugs and getting the colour's working for check solution