Refactor imageGenerator.java to ImageGenerator.java