Split .py files and add neural net