Clean up main.py so it is usable as an example

[lene]

• rename the parameters and functions to reflect what they really do
• add some docstrings too
• clean up the messy code
• add an example for saving and restoring trained networks
• move the code into an examples subdir and only keep main and argument parsing in main.py