Refactor SNAT code