Add genome2ai function