Improving 'prompt' function