Added a database wrapper class