minor updates for episode 5