Added check for storyline puzzle