diff --git a/test/test.ts b/test/test.ts
index 40eb79a..9986e30 100644
--- a/test/test.ts
+++ b/test/test.ts
@@ -10,11 +10,17 @@ tap.test('should create a valid helium class', async () => {
 
 tap.test('should create and update a wallet', async () => {
   const wallet = await testHelium.addWalletByAddress('13L7By1BmboTx1hRUuN3MtMjNmTFaWj5C6EbRZCK2sceaRCWZev');
+  expect(wallet).toBeInstanceOf(helium.HeliumWallet);
   console.log(wallet.earningsHnt24h);
   console.log(wallet.earningsHnt7d);
   console.log(wallet.earningsHnt14d);
   console.log(wallet.earningsHnt30d);
   console.log(wallet.hotspots);
+
+  expect(wallet.hotspots[0]).toBeInstanceOf(helium.HeliumHotspot);
+
+  // you can call wallet.update() at any time to update all wallet information
+  // also wallet.hotspots[0].update() will only update an individual hotspot;
 })
 
 tap.start();
diff --git a/ts/index.ts b/ts/index.ts
index 44391a1..6477952 100644
--- a/ts/index.ts
+++ b/ts/index.ts
@@ -1 +1,3 @@
-export * from './helium.classes.helium';
\ No newline at end of file
+export * from './helium.classes.helium';
+export * from './helium.classes.heliumwallet';
+export * from './helium.classes.hotspot';
\ No newline at end of file