Add code to enable ufw