added two test classes