Optimization runs properly, but cost ends too high (converges at ~4)