Minor fixes to model building notebook