Added README & made some improvements to code.