Tried to make UI a bit sexier