Add in more dying animation and logic