added difficulty & restarting