Refactor import for new guttmanlab-core