Added logic to prompt funciton, working on boolean function/ while loop