HAHA! fixed some annoying ui bugs