Fixes cube wonkyness

Wonkyness was due to height property being removed before animation
completed.

Now it is done 0.5s after animation is complete.