CS 6233 Final Project code for Abhimanyu Ghosh