If the AI tool can fetch related classes for the code that I'm working with that would be so helpful!