Tweaked the modelarchitecture file based on your comments