Fix comment for problem 20
Added solution for problem 21