next step is to add animation code to the below