Midterm project code WIP