Difficulty screen now works and starts game on chosen difficulty