updated code based on Chris's video - fixed do/end error