clean and update base.py -> work on docstring