Made the button animation a bit more correct, but it is still a long way from being right