Fix docstrings; Remove bugs in model loading