Minor adjustment to firewall script