To Do Update
Small difficulty selection fix