Adding code to the week2 thing