Added base models and setup.py