Update after getting animation to work