Updated class name to reflect new class name in Java
Reworked import