Done with data import and preprocessing
Done with peer assignment 2