Add docstrings
remove unnecessary models.py