Setup for problem 1