Added main guard