First draft of network with pickling
Add first trained network