Problem 4 finish