Edits to objective function