Update animation demo so that it functions correctly