The class file has been added to handel the solution