Update db.py

Adding schema for database.