Updates to controller and model for basic database setup.