Fixed some errors in episode 4