added setup for homework 3