MAINT: refactor create_flat_list

- Entities are now collected in a set (list previously).
- The function is better tested.
- The usage now longer requires to add the initial list to flat
  before hand.