HW2 starter code
import necessary packages and two datasets