Added difficulty selection screen