import data + successful solution for hw2.2