Refactor codes for ns scale