Added while loop and working on the story part of the project