Updated functions to the new datajoint 2.0 struct