added some docstrings and started test file for main models