Upadted the base class.