Added DocStrings

Added DocStrings to the python code. Fixed a few bugs along the way.