Documentation improvements + added __all__ variables