Improve setting up of the firewall rules