added a test case to the main class