First iteration of python rocchio code