mitb_cs604_final_project