added colors! and other UI improvements