Add source_generator