Adding injury logic