Displaying problems only in selected difficulty would be a very good feature and only needs a little modification to your code.