add function to set colour of LED