Started adding prototypical network complexities, almost done.