Method to set difficulty