diff --git a/add_company_modal.php b/add_company_modal.php
index d55e38b8..527188a1 100644
--- a/add_company_modal.php
+++ b/add_company_modal.php
@@ -61,12 +61,12 @@
           </div>
 
           <div class="form-group">
-            <label>Country</label>
+            <label>Country <strong class="text-danger">*</strong></label>
             <div class="input-group">
               <div class="input-group-prepend">
                 <span class="input-group-text"><i class="fa fa-fw fa-flag"></i></span>
               </div>
-              <select class="form-control select2" name="country">
+              <select class="form-control select2" name="country" required>
                 <option value="">- Country -</option>
                 <?php foreach($countries_array as $country_name) { ?>
                 <option><?php echo $country_name; ?></option>
diff --git a/alerts.php b/alerts.php
index d2bc3d6b..d08e31b1 100644
--- a/alerts.php
+++ b/alerts.php
@@ -10,7 +10,7 @@ $sql = mysqli_query($mysqli,"SELECT * FROM alerts WHERE alert_ack_date IS NULL A
   <div class="card-header py-2">
     <h3 class="card-title mt-2"><i class="fa fa-fw fa-exclamation-triangle"></i> Alerts</h3>
     <div class="card-tools">
-      <a href="post.php?ack_all_alerts" class="btn btn-success"> <i class="fa fa-check"></i> Acknowledge All</a>
+      <a href="post.php?ack_all_alerts" class="btn btn-primary"> <i class="fa fa-check"></i> Acknowledge All</a>
       <a href="alerts_archived.php" class="btn btn-secondary">Archived</a>
     </div>
   </div>
diff --git a/post.php b/post.php
index 630d6916..3f4936fe 100644
--- a/post.php
+++ b/post.php
@@ -457,6 +457,9 @@ if(isset($_POST['add_company'])){
 
     mysqli_query($mysqli,"INSERT INTO categories SET category_name = 'Service', category_type = 'Income', category_color = 'blue', category_created_at = NOW(), company_id = $company_id");
 
+    mysqli_query($mysqli,"INSERT INTO categories SET category_name = 'Friend', category_type = 'Refferal', category_color = 'blue', category_created_at = NOW(), company_id = $company_id");
+    mysqli_query($mysqli,"INSERT INTO categories SET category_name = 'Search Engine', category_type = 'Refferal', category_color = 'red', category_created_at = NOW(), company_id = $company_id");
+
     mysqli_query($mysqli,"INSERT INTO categories SET category_name = 'Cash', category_type = 'Payment Method', category_color = 'blue', category_created_at = NOW(), company_id = $company_id");
     mysqli_query($mysqli,"INSERT INTO categories SET category_name = 'Check', category_type = 'Payment Method', category_color = 'red', category_created_at = NOW(), company_id = $company_id");
 
diff --git a/setup.php b/setup.php
index 6ce78193..b5fa63e5 100644
--- a/setup.php
+++ b/setup.php
@@ -538,6 +538,9 @@ if(isset($_POST['add_company_settings'])){
 
   mysqli_query($mysqli,"INSERT INTO categories SET category_name = 'Service', category_type = 'Income', category_color = 'blue', category_created_at = NOW(), company_id = $company_id");
 
+  mysqli_query($mysqli,"INSERT INTO categories SET category_name = 'Friend', category_type = 'Refferal', category_color = 'blue', category_created_at = NOW(), company_id = $company_id");
+  mysqli_query($mysqli,"INSERT INTO categories SET category_name = 'Search Engine', category_type = 'Refferal', category_color = 'red', category_created_at = NOW(), company_id = $company_id");
+
   mysqli_query($mysqli,"INSERT INTO categories SET category_name = 'Cash', category_type = 'Payment Method', category_color = 'blue', category_created_at = NOW(), company_id = $company_id");
   mysqli_query($mysqli,"INSERT INTO categories SET category_name = 'Check', category_type = 'Payment Method', category_color = 'red', category_created_at = NOW(), company_id = $company_id");