Fix styles

src/State/ResourceGrabber.ts

@@ -1,13 +1,13 @@
 import { StateGrabber } from './StateGrabber';
 import { grabActiveVillageId } from '../Page/VillageBlock';
-import { grabResources } from '../Page/ResourcesBlock';
+import { grabResources, grabResourceStorage } from '../Page/ResourcesBlock';
 import { VillageState } from './VillageState';
 
 export class ResourceGrabber extends StateGrabber {
     grab(): void {
         const villageId = grabActiveVillageId();
-        const resources = grabResources();
         const state = new VillageState(villageId);
-        state.storeResources(resources);
+        state.storeResources(grabResources());
+        state.storeResourceStorage(grabResourceStorage());
     }
 }

src/State/VillageState.ts

@@ -1,5 +1,5 @@
 import { DataStorage } from '../Storage/DataStorage';
-import { Resources } from '../Game';
+import { Resources, ResourceStorage } from '../Game';
 
 export class VillageState {
     private storage: DataStorage;
@@ -13,7 +13,17 @@ export class VillageState {
 
     getResources(): Resources {
         let plain = this.storage.get('res');
-        let res = new Resources(0, 0, 0, 0, 0, 0);
+        let res = new Resources(0, 0, 0, 0);
         return Object.assign(res, plain) as Resources;
     }
+
+    storeResourceStorage(storage: ResourceStorage) {
+        this.storage.set('cap', storage);
+    }
+
+    getResourceStorage(): ResourceStorage {
+        let plain = this.storage.get('res');
+        let res = new ResourceStorage(0, 0);
+        return Object.assign(res, plain) as ResourceStorage;
+    }
 }