Add part 2 of problem nr 2