Initial setup of assignment 2