diff --git a/.idea/gradle.xml b/.idea/gradle.xml
index 11ee67b..361f3f1 100644
--- a/.idea/gradle.xml
+++ b/.idea/gradle.xml
@@ -15,6 +15,7 @@
             <option value="$PROJECT_DIR$/fancytoastlib" />
           </set>
         </option>
+        <option name="resolveModulePerSourceSet" value="false" />
       </GradleProjectSettings>
     </option>
   </component>
diff --git a/.idea/modules.xml b/.idea/modules.xml
index eec0c55..695c692 100644
--- a/.idea/modules.xml
+++ b/.idea/modules.xml
@@ -4,13 +4,7 @@
     <modules>
       <module fileurl="file://$PROJECT_DIR$/.idea/modules/FancyToast-Android.iml" filepath="$PROJECT_DIR$/.idea/modules/FancyToast-Android.iml" />
       <module fileurl="file://$PROJECT_DIR$/.idea/modules/app/FancyToast-Android.app.iml" filepath="$PROJECT_DIR$/.idea/modules/app/FancyToast-Android.app.iml" />
-      <module fileurl="file://$PROJECT_DIR$/.idea/modules/app/FancyToast-Android.app.androidTest.iml" filepath="$PROJECT_DIR$/.idea/modules/app/FancyToast-Android.app.androidTest.iml" />
-      <module fileurl="file://$PROJECT_DIR$/.idea/modules/app/FancyToast-Android.app.main.iml" filepath="$PROJECT_DIR$/.idea/modules/app/FancyToast-Android.app.main.iml" />
-      <module fileurl="file://$PROJECT_DIR$/.idea/modules/app/FancyToast-Android.app.unitTest.iml" filepath="$PROJECT_DIR$/.idea/modules/app/FancyToast-Android.app.unitTest.iml" />
       <module fileurl="file://$PROJECT_DIR$/.idea/modules/fancytoastlib/FancyToast-Android.fancytoastlib.iml" filepath="$PROJECT_DIR$/.idea/modules/fancytoastlib/FancyToast-Android.fancytoastlib.iml" />
-      <module fileurl="file://$PROJECT_DIR$/.idea/modules/fancytoastlib/FancyToast-Android.fancytoastlib.androidTest.iml" filepath="$PROJECT_DIR$/.idea/modules/fancytoastlib/FancyToast-Android.fancytoastlib.androidTest.iml" />
-      <module fileurl="file://$PROJECT_DIR$/.idea/modules/fancytoastlib/FancyToast-Android.fancytoastlib.main.iml" filepath="$PROJECT_DIR$/.idea/modules/fancytoastlib/FancyToast-Android.fancytoastlib.main.iml" />
-      <module fileurl="file://$PROJECT_DIR$/.idea/modules/fancytoastlib/FancyToast-Android.fancytoastlib.unitTest.iml" filepath="$PROJECT_DIR$/.idea/modules/fancytoastlib/FancyToast-Android.fancytoastlib.unitTest.iml" />
     </modules>
   </component>
 </project>
\ No newline at end of file
diff --git a/fancytoastlib/src/main/java/com/shashank/sony/fancytoastlib/FancyToast.java b/fancytoastlib/src/main/java/com/shashank/sony/fancytoastlib/FancyToast.java
index ef7b7de..e5d95d3 100644
--- a/fancytoastlib/src/main/java/com/shashank/sony/fancytoastlib/FancyToast.java
+++ b/fancytoastlib/src/main/java/com/shashank/sony/fancytoastlib/FancyToast.java
@@ -18,10 +18,12 @@
 import java.lang.annotation.RetentionPolicy;
 
 public class FancyToast extends Toast {
-    
+
     @Retention(RetentionPolicy.SOURCE)
     @IntDef({SUCCESS, WARNING, ERROR, INFO, DEFAULT, CONFUSING})
-    public @interface LayoutType {}
+    public @interface LayoutType {
+    }
+
     public static final int SUCCESS = 1;
     public static final int WARNING = 2;
     public static final int ERROR = 3;
@@ -30,9 +32,11 @@ public class FancyToast extends Toast {
     public static final int CONFUSING = 6;
 
 
-    @IntDef({ LENGTH_SHORT, LENGTH_LONG })
+    @IntDef({LENGTH_SHORT, LENGTH_LONG})
     @Retention(RetentionPolicy.SOURCE)
-    public @interface Duration {}
+    public @interface Duration {
+    }
+
     public static final int LENGTH_SHORT = Toast.LENGTH_SHORT;
     public static final int LENGTH_LONG = Toast.LENGTH_LONG;
 
@@ -42,7 +46,7 @@ public class FancyToast extends Toast {
      * <p>
      *
      * @param context The context to use.  Usually your {@link Application}
-     * or {@link Activity} object.
+     *                or {@link Activity} object.
      */
     public FancyToast(@NonNull Context context) {
         super(context);
@@ -117,14 +121,10 @@ public static Toast makeText(@NonNull Context context, CharSequence message, @Du
             case INFO:
                 linearLayout.setBackgroundResource(R.drawable.info_shape);
                 break;
-            case DEFAULT: {
-                linearLayout.setBackgroundResource(R.drawable.default_shape);
-                img.setVisibility(View.GONE);
-                break;
-            }
             case CONFUSING:
                 linearLayout.setBackgroundResource(R.drawable.confusing_shape);
                 break;
+            case DEFAULT:
             default:
                 linearLayout.setBackgroundResource(R.drawable.default_shape);
                 img.setVisibility(View.GONE);