Added ability to change plot colour