UI update

Still sucks XD