Small tweaks to fit code, had a bug in partial_fit iterator