Updated game to match changes with colour class