feat(ClassGenerator): Add a function to get sub directories