Refactoring - annotation now is a function.
Changing colours for the graphs