Support HMR for class components

This is a long overdue and should fix some issues people had with HMR when migrating from CRA.