Fixed generator and updated readme