make heroes an attr instead of a method
handle edge cases