Add generator class script