Custom entity types?

This module looks incredibly useful but it isn't picking up my custom entity types. Is there a way to plug into the module so that my entity types appear in the UI?

Thanks!