Added a note enum and a function to get the color