[TASK] Cleanup EM Code