Simple neural net implementation in Julia