added a restart button and removed unused methods