Updated the problem statement