Added restore into prompt