[FEATURE] Add 6.0 compatibility

In 6.0 a new argument ``id`` is introduced that will replace ``key``