Updated the controller to include the spree module
Updated Version