cleaned up library

added colour function