Updated README.md and resubmit work from TA's comments