Updated the Docstrings. Added balance functionality.