diff --git a/src/main/java/org/onosproject/yangutils/translator/tojava/JavaFileInfo.java b/src/main/java/org/onosproject/yangutils/translator/tojava/JavaFileInfo.java
index 3024a47..3a53884 100644
--- a/src/main/java/org/onosproject/yangutils/translator/tojava/JavaFileInfo.java
+++ b/src/main/java/org/onosproject/yangutils/translator/tojava/JavaFileInfo.java
@@ -53,7 +53,7 @@
      * definition.
      *
      * @return the types of files being generated corresponding to the YANG
-     *         definition
+     * definition
      */
     public int getGeneratedFileTypes() {
         return genFileTypes;
@@ -64,13 +64,24 @@
      * definition.
      *
      * @param fileTypes the types of files being generated corresponding to the
-     *            YANG definition
+     * YANG definition
      */
     public void setGeneratedFileTypes(int fileTypes) {
         genFileTypes = fileTypes;
     }
 
     /**
+     * Adds the types of files being generated corresponding to the YANG
+     * definition.
+     *
+     * @param fileTypes the types of files being generated corresponding to the
+     * YANG definition
+     */
+    public void addGeneratedFileTypes(int fileTypes) {
+        genFileTypes |= fileTypes;
+    }
+
+    /**
      * Returns the java name of the node.
      *
      * @return the java name of node