fixing more bugs - ui stuff