Updating the code to match the most recent tut