Add main guard