Updated tests, added more methods to hero class