Updating readme and code