Add difficulty to level generator