Updating Controller.java