minor fixes for episode 5