Working on basic generator and model code