DATA1030_final_project