Modifications for episode 4