adding while, do functionality, working on if statements