Add difficulty too