added a line of code to make story flow better