Added the code for episode 9