Add graphics for the leaves and improve animation