[+] documentation and refactoring for the class for generation