Install UFW

sudo apt-get install ufw

Set default rules

sudo ufw default allow outgoing
sudo ufw default deny incoming

Add rules

sudo ufw allow SSH_PORT
sudo ufw allow 80/tcp
sudo ufw allow 443/tcp

Enable firewall

sudo ufw enable

Useful command

sudo ufw status