Setting up code given to us