smaller data set and code complete