Added class file and created class as well as the init function