removed downloading part so that the script can use a pickled dataset