Minor edits to hero unit