remove unused main method

test cases updates