Update Controller.java

Testing screen that I'm working on right now