updated setup for new version
added docstrings and improved readability
fixed pep8 errors
updated readme
updated readme