Added int choice and difficulty()