Update 02_bank.sol

modifier onlyOwner
xanpena · Jan 15, 2022 · 2a6ae1e · 2a6ae1e
1 parent d5b6216
commit 2a6ae1e
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/02_bank.sol b/02_bank.sol
@@ -5,6 +5,10 @@ contract Bank {
 
     address owner;
 
+    modifier onlyOwner {
+        require(msg.sender == owner);
+    }
+
     function Bank() payable public {
         owner = msg.sender;
     }
@@ -13,8 +17,7 @@ contract Bank {
         require(msg.value == amount);
     }
 
-    function withdrawBalance() public {
-        require(msg.sender == owner);
+    function withdrawBalance() public onlyOwner {
         msg.sender.transfer(address(this).balance);
     }
 }