Updated problem statement