Switched from distutils to setuptools and added entrypoint and some small changes