- if always_reveal_states == True or self.types_revealed[i] == True:
- img = small_images[self.colour][self.types[i]]
+ if always_reveal_states == True or self.types[i][0] != '?':
+ if self.types[i][0] == '?':
+ img = small_images[self.colour][self.types[i][1:]]
+ else:
+ img = small_images[self.colour][self.types[i]]