Added header and created class Mage