added this stuff
Add new class