Updating some functionality in the unit testing class
Updated the assertContainsOnly functionality