Add hooks to class components with higher-order components