Change prompt if/else logic