update the code
Add in attack and die effect