Added feature to change difficulty