Add database wrapper