add the code we wrote in class