Removing pybrain; writing in vanilla python