Added damage tracking