Added code for Episode 8