Added setup.py and reorganized.
Fixed some docstrings.