1) Add the Controller class
2) Edit the DBConnector class