Added set methods to allow changing the class