Improve DBConnection class