Fixed bug + added example of trained network