Added getConnection() method