Fixed issue with screen.initialize never being called.
Also added difficulty selection screen.