Added in promt and if statement