Added class structure and functions to generate files