re-add back in animation code