small ui fixes & changes