Updated the controller with proper DB calls.