Minor changes to incorporate pyBrain