This is a follow up for #14 :

The return type specified here is list, but in fact, we return a dictionary.

The dictionary contains vectors as values (supposedly), but those have shape (1, n) for n labels.

The dictionary also contains measurements for background. This should be avoided.

I should clean up this mess :-)