Updated README.md, modularised parts of the code.
* TODO: More modularisation, commenting