Small update, use os agrument for port