Add comment for problem's difficulty