Added ability to change difficulty in menu