Added some test coverages for the classes created