Add difficulty menu
Some bug fixes