Generalize code

Create functions for main NN training and data import (second commit) Expand options for import (third commit)

Hopefully it's easy enough to see what I did by looking at each commit separately (and maybe turning off whitespace changes)