Additional plotting + first versions of neural networks