Updated README and trained optimal NN