code we made in lecture