import data + successful solution for hw4.4