Final code for Final Project

Might need to add some comments