Add generator (Not working yet)