diff --git a/assembly.xml b/assembly.xml
index 7d15b76..e5b7ce4 100644
--- a/assembly.xml
+++ b/assembly.xml
@@ -13,7 +13,7 @@
             <directory>${project.basedir}/src/main/resources</directory>
             <outputDirectory>config</outputDirectory>
             <includes>
-                <include>*/**</include>
+                <include>*.yml</include>
             </includes>
         </fileSet>
 
diff --git a/bin/start.bat b/bin/start.bat
index 1606024..8d5c869 100644
--- a/bin/start.bat
+++ b/bin/start.bat
@@ -4,4 +4,5 @@ set JAVA_CMD=%JAVA_HOME%\bin\java
 set JAVA_OPTS=-Xmx512m -Xms512m -Xmn256m -Xss256k
 set CONFIG_FILE=../config/application.yml
 set TARGET=../lib/kafka-console-ui.jar
-%JAVA_CMD% -jar %TARGET% --spring.config.location=%CONFIG_FILE%
\ No newline at end of file
+set DATA_DIR=..
+%JAVA_CMD% -jar %TARGET% --spring.config.location=%CONFIG_FILE% --data.dir=%DATA_DIR%
\ No newline at end of file
diff --git a/pom.xml b/pom.xml
index 72053fe..6bb1a64 100644
--- a/pom.xml
+++ b/pom.xml
@@ -155,26 +155,26 @@
                         </executions>
                     </plugin>
 
-                    <plugin>
-                        <groupId>org.codehaus.mojo</groupId>
-                        <artifactId>build-helper-maven-plugin</artifactId>
-                        <version>3.2.0</version>
-                        <executions>
-                            <execution>
-                                <id>add-source</id>
-                                <phase>generate-sources</phase>
-                                <goals>
-                                    <goal>add-source</goal>
-                                </goals>
-                                <configuration>
-                                    <sources>
-                                        <source>src/main/java</source>
-                                        <source>src/main/scala</source>
-                                    </sources>
-                                </configuration>
-                            </execution>
-                        </executions>
-                    </plugin>
+<!--                    <plugin>-->
+<!--                        <groupId>org.codehaus.mojo</groupId>-->
+<!--                        <artifactId>build-helper-maven-plugin</artifactId>-->
+<!--                        <version>3.2.0</version>-->
+<!--                        <executions>-->
+<!--                            <execution>-->
+<!--                                <id>add-source</id>-->
+<!--                                <phase>generate-sources</phase>-->
+<!--                                <goals>-->
+<!--                                    <goal>add-source</goal>-->
+<!--                                </goals>-->
+<!--                                <configuration>-->
+<!--                                    <sources>-->
+<!--                                        <source>src/main/java</source>-->
+<!--                                        <source>src/main/scala</source>-->
+<!--                                    </sources>-->
+<!--                                </configuration>-->
+<!--                            </execution>-->
+<!--                        </executions>-->
+<!--                    </plugin>-->
 
                     <plugin>
                         <artifactId>maven-compiler-plugin</artifactId>