Added exit and restart buttons