bugfix
Remove clean up function.
Added colours.