Updated hero controller