Cleared up comment on cost_function