Assignment 6 set up