Minor changes to hero class