Modify work_orders_controller