Added a new class for add