Better example code for models using chains framework (#1347)

Better example code for models using chains framework (#1347) #2