fixed restart functionality and added again