Add new functionality for the masstransit framework