Updated readme & added some comments etc