Initially added helper classes for generator

#8 Generator refactoring