Completed the code for the CustomerController