import data + successful solution for hw2.1