Add some animations and small UI fixes