Added __all__
Fixed comments
Refined code
Added module comment
Changed assert to assert_equal or assert_true