Updated readme and minor fix