feat: test dataset

Closes #3
feat: save trained neural network

Closes #5