Add difficulty check