Fixed "Press enter" and difficulty function