Work on restructuring train_superimposed.py a bit.

Currently not working.