diff --git a/setup.py b/setup.py
index 0859fe3..7195da6 100644
--- a/setup.py
+++ b/setup.py
@@ -1,3 +1,6 @@
+#!/usr/bin/env python
+# coding: utf-8
+
 import os
 from setuptools import setup, find_packages
 
@@ -11,7 +14,7 @@ def read(fname):
     "scipy",
     "pandas",
     "matplotlib",
-    "sklearn",
+    "scikit-learn",
     "pyyaml>=5.1",
     "pytorch-lightning",
     "decorator",
@@ -37,12 +40,12 @@ def read(fname):
     long_description_content_type="text/markdown",
     license="Apache License, Version 2.0",
     keywords=[
-        "machine learning",
+        "Machine Learning",
         "MLOps",
         "Pytorch",
         "PytorchLightning",
         "Lightning",
-        "pipeline"
+        "Pipeline"
     ],
     url="https://github.com/murnanedaniel/train-track",
 )