Added magnesium atom
Added dummy states to allow simulation
removed unused import