diff --git a/setup.py b/setup.py
index e1e0f53..b349f68 100644
--- a/setup.py
+++ b/setup.py
@@ -16,11 +16,15 @@
     version=version,
     author="Emil Zakirov",
     author_email="bonlimezak@gmail.com",
-    packages=find_packages(exclude=["test", "docs", "examples"]),
+    packages=find_packages(exclude=["tests", "docs", "examples"]),
     url="https://github.com/bonlime/pytorch-tools",
     description="Tool box for PyTorch",
     long_description=readme,
-    classifiers=["Programming Language :: Python :: 3.6",],
+    long_description_content_type="text/markdown",
+    classifiers=[
+        "Programming Language :: Python :: 3.6",
+        "Programming Language :: Python :: 3.7",
+    ],
     setup_requires=["setuptools_scm"],
     python_requires=">=3, <4",
     install_requires=requirements,