added the class.. 2