reflect changes in setup.py

Author: bexxmodd

setup.py

@@ -8,15 +8,19 @@
 
 setup(
     name='vizex',
-    version='2.1.0',
+    version='2.1.1',
     author='Beka Modebadze',
     author_email='bexx.modd@gmail.com',
     description='UNIX/Linux Terminal program to graphically display the disk space usage and/or directory data',
     long_description=long_description,
     long_description_content_type ='text/markdown',
     url='https://github.com/bexxmodd/vizex',
     package_dir = {'': 'vizex'},
-    py_modules=['cli',  'vizexdu/disks', 'tools', 'vizexdu/charts', 'vizexdu/battery', 'vizexdu/cpu', 'vizexdf/files'],
+    py_modules=[
+        'cli', 'vizexdu/disks', 'tools',
+        'vizexdu/charts', 'vizexdu/battery', 'vizexdu/cpu',
+        'vizexdf/files', 'vizextree/viztree'
+    ],
     packages = find_packages(where='vizex'),
     classifiers=[
         "Programming Language :: Python :: 3.8",