Add init prompt function