Adding while statement and True/False