I played around with the values and changed colours.
Started the code as a class