Rewrote the problem definition