First try at applying pybrain