added difficulty switch; working on diff-2