changes to the __init__ method
changes to sample
FIrst pass through the full model