Added error check when a user enters an invalid difficulty.