Fixed: Clean up Python unit tests
Documentation: Commenting for clarification