fixed code

needed to not download all of the data.  Skip that step for the
assignment.