2 line added to class Heroes