Improved setup.py, added colour coding to UI