Added more unit  tests and refactored comments to comply with PEP 8 standards.