Added difficulty to the main screen