moved data extraction from assignment 1 to assignment_1