UI Enhancement

Yeah... just like that