Update problem numbers in comments