Add damage calculation