Working models file, will acually hash out models.py file tomorrow