Setting up problem 17 framework