added to code from video 18