diff --git a/.idea/modules.xml b/.idea/modules.xml
index 97a9141..4a01aad 100644
--- a/.idea/modules.xml
+++ b/.idea/modules.xml
@@ -2,9 +2,11 @@
 <project version="4">
   <component name="ProjectModuleManager">
     <modules>
+      <module fileurl="file://$PROJECT_DIR$/Sample/Sample.iml" filepath="$PROJECT_DIR$/Sample/Sample.iml" />
       <module fileurl="file://$PROJECT_DIR$/Sample/Sample.iml" filepath="$PROJECT_DIR$/Sample/Sample.iml" />
       <module fileurl="file://$PROJECT_DIR$/Space-Navigation-View.iml" filepath="$PROJECT_DIR$/Space-Navigation-View.iml" />
       <module fileurl="file://$PROJECT_DIR$/spacelib/spacelib.iml" filepath="$PROJECT_DIR$/spacelib/spacelib.iml" />
+      <module fileurl="file://$PROJECT_DIR$/spacelib/spacelib.iml" filepath="$PROJECT_DIR$/spacelib/spacelib.iml" />
     </modules>
   </component>
 </project>
\ No newline at end of file
diff --git a/.idea/vcs.xml b/.idea/vcs.xml
new file mode 100644
index 0000000..35eb1dd
--- /dev/null
+++ b/.idea/vcs.xml
@@ -0,0 +1,6 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="VcsDirectoryMappings">
+    <mapping directory="" vcs="Git" />
+  </component>
+</project>
\ No newline at end of file
diff --git a/Sample/src/main/AndroidManifest.xml b/Sample/src/main/AndroidManifest.xml
index 046a5d0..1ba604f 100644
--- a/Sample/src/main/AndroidManifest.xml
+++ b/Sample/src/main/AndroidManifest.xml
@@ -15,6 +15,8 @@
                 <category android:name="android.intent.category.LAUNCHER" />
             </intent-filter>
         </activity>
+
+        <activity android:name=".ActivityWithBadge"></activity>
     </application>
 
 </manifest>
\ No newline at end of file
diff --git a/Sample/src/main/java/com/luseen/spacenavigationview/ActivityWithBadge.java b/Sample/src/main/java/com/luseen/spacenavigationview/ActivityWithBadge.java
new file mode 100644
index 0000000..538001c
--- /dev/null
+++ b/Sample/src/main/java/com/luseen/spacenavigationview/ActivityWithBadge.java
@@ -0,0 +1,121 @@
+package com.luseen.spacenavigationview;
+
+import android.graphics.Color;
+import android.os.Bundle;
+import android.support.v4.content.ContextCompat;
+import android.support.v7.app.AppCompatActivity;
+import android.support.v7.widget.GridLayoutManager;
+import android.support.v7.widget.RecyclerView;
+import android.util.Log;
+import android.view.View;
+import android.widget.Button;
+import android.widget.Toast;
+
+import com.luseen.spacenavigation.SpaceItem;
+import com.luseen.spacenavigation.SpaceNavigationView;
+import com.luseen.spacenavigation.SpaceOnClickListener;
+import com.luseen.spacenavigation.SpaceOnLongClickListener;
+
+import java.util.ArrayList;
+import java.util.List;
+
+
+public class ActivityWithBadge extends AppCompatActivity {
+
+    private SpaceNavigationView spaceNavigationView;
+
+    @Override
+    protected void onCreate(Bundle savedInstanceState) {
+        super.onCreate(savedInstanceState);
+        setContentView(R.layout.activity_badge);
+
+        Button btnShowBadge = (Button) findViewById(R.id.btnBadge);
+        btnShowBadge.setOnClickListener(new View.OnClickListener() {
+            @Override
+            public void onClick(View v) {
+                spaceNavigationView.shouldShowFullBadgeText(true);
+                spaceNavigationView.showBadgeAtIndex(0, 2, Color.RED);
+                spaceNavigationView.showBadgeAtIndex(1, 3, Color.CYAN);
+                spaceNavigationView.showBadgeAtIndex(2, 4, Color.MAGENTA);
+                spaceNavigationView.showBadgeAtIndex(3, 23, Color.BLUE);
+            }
+        });
+
+        spaceNavigationView = (SpaceNavigationView) findViewById(R.id.space);
+        spaceNavigationView.initWithSaveInstanceState(savedInstanceState);
+        spaceNavigationView.addSpaceItem(new SpaceItem("HOME", R.drawable.account));
+        spaceNavigationView.addSpaceItem(new SpaceItem("SEARCH", R.drawable.magnify));
+        spaceNavigationView.addSpaceItem(new SpaceItem("HOME", R.drawable.account));
+        spaceNavigationView.addSpaceItem(new SpaceItem("SEARCH", R.drawable.magnify));
+        spaceNavigationView.shouldShowFullBadgeText(false);
+        spaceNavigationView.setCentreButtonIconColorFilterEnabled(false);
+
+        spaceNavigationView.setSpaceOnClickListener(new SpaceOnClickListener() {
+            @Override
+            public void onCentreButtonClick() {
+                Log.d("onCentreButtonClick ", "onCentreButtonClick");
+            }
+
+            @Override
+            public void onItemClick(int itemIndex, String itemName) {
+                Log.d("onItemClick ", "" + itemIndex + " " + itemName);
+            }
+
+            @Override
+            public void onItemReselected(int itemIndex, String itemName) {
+                Log.d("onItemReselected ", "" + itemIndex + " " + itemName);
+            }
+        });
+
+        spaceNavigationView.setSpaceOnLongClickListener(new SpaceOnLongClickListener() {
+            @Override
+            public void onCentreButtonLongClick() {
+                Toast.makeText(ActivityWithBadge.this, "onCentreButtonLongClick", Toast.LENGTH_SHORT).show();
+            }
+
+            @Override
+            public void onItemLongClick(int itemIndex, String itemName) {
+                Toast.makeText(ActivityWithBadge.this, itemIndex + " " + itemName, Toast.LENGTH_SHORT).show();
+            }
+        });
+        spaceNavigationView.showIconOnly();
+        setUpRecyclerView();
+    }
+
+    @Override
+    protected void onSaveInstanceState(Bundle outState) {
+        super.onSaveInstanceState(outState);
+        spaceNavigationView.onSaveInstanceState(outState);
+    }
+
+    private void setUpRecyclerView() {
+        RecyclerView recyclerView = (RecyclerView) findViewById(R.id.recyclerView);
+        RecyclerAdapter adapter = new RecyclerAdapter(dummyStrings());
+        recyclerView.setHasFixedSize(true);
+        recyclerView.setLayoutManager(new GridLayoutManager(this, 2));
+        recyclerView.setAdapter(adapter);
+        adapter.setRecyclerClickListener(new RecyclerAdapter.RecyclerClickListener() {
+            @Override
+            public void onClick(int position) {
+                if (position == 0) {
+                    spaceNavigationView.showBadgeAtIndex(1, 54, ContextCompat.getColor(ActivityWithBadge.this, R.color.badge_background_color));
+                } else if (position == 1) {
+                    spaceNavigationView.hideBudgeAtIndex(1);
+                }
+            }
+        });
+    }
+
+    private List<String> dummyStrings() {
+        List<String> colorList = new ArrayList<>();
+        colorList.add("#354045");
+        colorList.add("#20995E");
+        colorList.add("#76FF03");
+        colorList.add("#E26D1B");
+        colorList.add("#911717");
+        colorList.add("#9C27B0");
+        colorList.add("#FFC107");
+        colorList.add("#01579B");
+        return colorList;
+    }
+}
\ No newline at end of file
diff --git a/Sample/src/main/java/com/luseen/spacenavigationview/MainActivity.java b/Sample/src/main/java/com/luseen/spacenavigationview/MainActivity.java
index f1f9417..bf7d0b7 100644
--- a/Sample/src/main/java/com/luseen/spacenavigationview/MainActivity.java
+++ b/Sample/src/main/java/com/luseen/spacenavigationview/MainActivity.java
@@ -1,5 +1,6 @@
 package com.luseen.spacenavigationview;
 
+import android.content.Intent;
 import android.os.Bundle;
 import android.support.v4.content.ContextCompat;
 import android.support.v7.app.AppCompatActivity;
@@ -26,6 +27,8 @@ protected void onCreate(Bundle savedInstanceState) {
         super.onCreate(savedInstanceState);
         setContentView(R.layout.activity_main);
 
+        Toast.makeText(getApplicationContext(), "Long press center button to show badge example", Toast.LENGTH_LONG).show();
+
         spaceNavigationView = (SpaceNavigationView) findViewById(R.id.space);
         spaceNavigationView.initWithSaveInstanceState(savedInstanceState);
         spaceNavigationView.addSpaceItem(new SpaceItem("HOME", R.drawable.account));
@@ -54,7 +57,9 @@ public void onItemReselected(int itemIndex, String itemName) {
         spaceNavigationView.setSpaceOnLongClickListener(new SpaceOnLongClickListener() {
             @Override
             public void onCentreButtonLongClick() {
-                Toast.makeText(MainActivity.this, "onCentreButtonLongClick", Toast.LENGTH_SHORT).show();
+//                Toast.makeText(MainActivity.this, "onCentreButtonLongClick", Toast.LENGTH_SHORT).show();
+                Intent intent = new Intent(MainActivity.this, ActivityWithBadge.class);
+                startActivity(intent);
             }
 
             @Override
diff --git a/Sample/src/main/res/layout/activity_badge.xml b/Sample/src/main/res/layout/activity_badge.xml
new file mode 100644
index 0000000..6185b5c
--- /dev/null
+++ b/Sample/src/main/res/layout/activity_badge.xml
@@ -0,0 +1,46 @@
+<?xml version="1.0" encoding="utf-8"?>
+<android.support.design.widget.CoordinatorLayout xmlns:android="http://schemas.android.com/apk/res/android"
+    xmlns:app="http://schemas.android.com/apk/res-auto"
+    xmlns:tools="http://schemas.android.com/tools"
+    android:id="@+id/main_content"
+    android:layout_width="match_parent"
+    android:layout_height="match_parent"
+    tools:context="com.luseen.spacenavigationview.MainActivity">
+
+    <LinearLayout
+        android:layout_width="match_parent"
+        android:layout_height="wrap_content"
+        android:orientation="vertical">
+
+        <TextView
+            android:layout_width="match_parent"
+            android:layout_height="wrap_content"
+            android:padding="20dp"
+            android:text="This library doesn't support loading of badges during initialisation. It takes some time for bottom navigation to get initialize. Hence I have given a button below to load badges on click. In production you can call the showBadgeAtIndex method after a network call or use it by making a thread call of 3 second" />
+
+        <Button
+            android:id="@+id/btnBadge"
+            android:layout_width="match_parent"
+            android:layout_height="wrap_content"
+            android:text="Click to Show Badge" />
+
+        <RelativeLayout
+            android:layout_width="match_parent"
+            android:layout_height="match_parent">
+
+            <android.support.v7.widget.RecyclerView
+                android:id="@+id/recyclerView"
+                android:layout_width="match_parent"
+                android:layout_height="match_parent" />
+
+            <com.luseen.spacenavigation.SpaceNavigationView
+                android:id="@+id/space"
+                android:layout_width="match_parent"
+                android:layout_height="wrap_content"
+                android:layout_alignParentBottom="true"
+                android:layout_gravity="bottom"
+                app:centre_part_linear="true"
+                app:layout_behavior="com.luseen.spacenavigation.SpaceNavigationViewBehavior" />
+        </RelativeLayout>
+    </LinearLayout>
+</android.support.design.widget.CoordinatorLayout>
\ No newline at end of file
diff --git a/build.gradle b/build.gradle
index d99d4f8..4099202 100644
--- a/build.gradle
+++ b/build.gradle
@@ -4,8 +4,8 @@ buildscript {
         jcenter()
     }
     dependencies {
-        classpath 'com.android.tools.build:gradle:2.3.0-beta3'
-        classpath 'com.novoda:bintray-release:0.3.4'
+        classpath 'com.android.tools.build:gradle:3.0.1'
+        classpath 'com.novoda:bintray-release:0.8.0'
     }
 }
 
diff --git a/gradle/wrapper/gradle-wrapper.properties b/gradle/wrapper/gradle-wrapper.properties
index c0c2bd8..18d7b73 100644
--- a/gradle/wrapper/gradle-wrapper.properties
+++ b/gradle/wrapper/gradle-wrapper.properties
@@ -1,6 +1,6 @@
-#Sun Jan 29 18:23:19 AMT 2017
+#Wed Mar 14 18:44:51 IST 2018
 distributionBase=GRADLE_USER_HOME
 distributionPath=wrapper/dists
 zipStoreBase=GRADLE_USER_HOME
 zipStorePath=wrapper/dists
-distributionUrl=https\://services.gradle.org/distributions/gradle-3.3-all.zip
+distributionUrl=https\://services.gradle.org/distributions/gradle-4.1-all.zip
diff --git a/spacelib/src/main/res/drawable/near_me.xml b/spacelib/src/main/res/drawable/near_me.xml
index d4e92fc..0b044d4 100644
--- a/spacelib/src/main/res/drawable/near_me.xml
+++ b/spacelib/src/main/res/drawable/near_me.xml
@@ -4,5 +4,5 @@
     android:width="24dp"
     android:viewportWidth="24"
     android:viewportHeight="24">
-    <path android:fillColor="#fff" android:pathData="M21,3L3,10.53V11.5L9.84,14.16L12.5,21H13.46L21,3Z" />
+    <path android:fillColor="#ff0000" android:pathData="M21,3L3,10.53V11.5L9.84,14.16L12.5,21H13.46L21,3Z" />
 </vector>
\ No newline at end of file
diff --git a/spacelib/src/main/res/values/dimens.xml b/spacelib/src/main/res/values/dimens.xml
index 332bcd5..eafbf8b 100644
--- a/spacelib/src/main/res/values/dimens.xml
+++ b/spacelib/src/main/res/values/dimens.xml
@@ -3,7 +3,7 @@
     <dimen name="main_content_height">50dp</dimen>
     <dimen name="space_centre_button_default_size">56dp</dimen>
     <dimen name="space_centre_button_margin">6dp</dimen>
-    <dimen name="centre_content_width">110dp</dimen>
+    <dimen name="centre_content_width">80dp</dimen>
     <dimen name="space_item_icon_default_size">18dp</dimen>
     <dimen name="space_item_icon_only_size">25dp</dimen>
     <dimen name="space_item_text_default_size">14sp</dimen>