Move main program to main.py, add some more example training sets and format better