UI improvements - add some colors and blue buttons