Adding test case instead of main class