added restart functionality