Changes to README.md and refining logic for project