Add electromagnet control