Improve architecture
Improve architecture by having all layers in the same list
Minor cleanup
Add some comments
Improve architecture to support more hidden layers
Revert training of multi hidden layers, was not working at all