JUNit instead of Main class