Problem 2: Code solves the problem but doesn't take inputs