Setup for homework 3