Added main to JUnit tests