Adding README file and tiding up the code