Which units would you like to replace or add?

Maybe he controls when to release the electric charge and when not. Alternatively, you could divide the tasks: let him attack large creatures with his fist and small creatures with his foot, and the electric charge won't be a hindrance.