Add db connection and models