Add doc string to problem 01