Add database integration