Added restart functionality