Add ability to fly. Only increase max energy by the amount actually gained, instead of the amount attempted to recover.