Modified the attack animation logic