Added db handling class