Update Controller.java

updating control with new logic