Simple python quiz code with score