Edits to problem statement