Add nicer prompt