Problem 2 finish