Code from episode 2.