From a9561d1694e354f2f536809151bf2db0b732d613 Mon Sep 17 00:00:00 2001
From: Rick Barrette <rickbarrette@gmail.com>
Date: Thu, 4 Aug 2016 00:02:56 -0400
Subject: [PATCH] Update vehicles_controller.rb

---
 app/controllers/vehicles_controller.rb | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/app/controllers/vehicles_controller.rb b/app/controllers/vehicles_controller.rb
index 2f96c87..bee5eb2 100644
--- a/app/controllers/vehicles_controller.rb
+++ b/app/controllers/vehicles_controller.rb
@@ -18,6 +18,12 @@ class VehiclesController < ApplicationController
   
   # display a list of all vehicles
   def index
+    begin
+      @vehicles = Customers.find_by_id(params[:id]).vehicles
+    rescue ActiveRecord::RecordNotFound
+      render_404
+    end
+    
     if params[:search]
       @vehicles = Vehicle.search(params[:search]).paginate(:page => params[:page])
       if only_one_non_zero?(@vehicles)