add prompt in main(unfinish)