Adding setup.py
Updating structure
Updating docstrings