Change the controller to use the "database"