Setting up for assignment 4