made fixes
Merge https://github.com/sachinravi14/COS-598C-Project