correction made to test2 class