Added colouring of cells based on section

This was the extra credit section of the challenge