update Controller.java