Added Mage entity to stages 1 and 2