Some minor readme and code updates