Db/setup

Added models and migrations. Needs some fixing but moving over to your repo