More UI fixes, first animation works