Add support for difficulty level