Added basic functionality for difficulty setting