Forgot to update the Edit and Delete methods.