Added "restart" ability