Updated readme and code documentation