[TASK] Cleanup EM update class