Colour rendering works
Tried to refactor but getting errors now.