src/main/java/rocks/zipcode/atm/CashMachine.java

@@ -46,6 +46,14 @@ public class CashMachine {
             );
         }
     }
+    public void balance(int amount) {
+        if (accountData != null) {
+            tryCall(
+                    () -> bank.balance(accountData, amount),
+                    update
+            );
+        }
+    }
 
     public void exit() {
         if (accountData != null) {
@@ -72,4 +80,6 @@ public class CashMachine {
             System.out.println("Error: " + e.getMessage());
         }
     }
+
+
 }

src/main/java/rocks/zipcode/atm/CashMachineApp.java

@@ -1,5 +1,6 @@
 package rocks.zipcode.atm;
 
+import javafx.scene.layout.HBox;
 import rocks.zipcode.atm.bank.Bank;
 import javafx.application.Application;
 import javafx.scene.Parent;
@@ -19,13 +20,15 @@ public class CashMachineApp extends Application {
     private TextField field = new TextField();
     private CashMachine cashMachine = new CashMachine(new Bank());
 
+
     private Parent createContent() {
         VBox vbox = new VBox(10);
-        vbox.setPrefSize(600, 600);
+        vbox.setPrefSize(800, 600);
 
         TextArea areaInfo = new TextArea();
 
-        Button btnSubmit = new Button("Set Account ID");
+        Button btnSubmit = new Button("Account ID");
+        btnSubmit.setStyle("-fx-border-color: #0000ff; -fx-border-width: 5px;");
         btnSubmit.setOnAction(e -> {
             int id = Integer.parseInt(field.getText());
             cashMachine.login(id);
@@ -34,6 +37,7 @@ public class CashMachineApp extends Application {
         });
 
         Button btnDeposit = new Button("Deposit");
+        btnDeposit.setStyle("-fx-border-color: #0000ff; -fx-border-width: 5px;");
         btnDeposit.setOnAction(e -> {
             int amount = Integer.parseInt(field.getText());
             cashMachine.deposit(amount);
@@ -42,6 +46,7 @@ public class CashMachineApp extends Application {
         });
 
         Button btnWithdraw = new Button("Withdraw");
+        btnWithdraw.setStyle("-fx-border-color: #0000ff; -fx-border-width: 5px;");
         btnWithdraw.setOnAction(e -> {
             int amount = Integer.parseInt(field.getText());
             cashMachine.withdraw(amount);
@@ -49,7 +54,17 @@ public class CashMachineApp extends Application {
             areaInfo.setText(cashMachine.toString());
         });
 
+        Button btnBalance = new Button("Balance");
+        btnBalance.setStyle("-fx-border-color: #0000ff; -fx-border-width: 5px;");
+        btnBalance.setOnAction(e -> {
+            int amount = Integer.parseInt(field.getText());
+            cashMachine.balance(amount);
+
+            areaInfo.setText(cashMachine.toString());
+        });
+
         Button btnExit = new Button("Exit");
+        btnExit.setStyle("-fx-border-color: #0000ff; -fx-border-width: 5px;");
         btnExit.setOnAction(e -> {
             cashMachine.exit();
 
@@ -68,8 +83,11 @@ public class CashMachineApp extends Application {
 
     @Override
     public void start(Stage stage) throws Exception {
+        stage.setTitle("Welcome to ZipCloudBank");
+
         stage.setScene(new Scene(createContent()));
         stage.show();
+
     }
 
     public static void main(String[] args) {

src/main/java/rocks/zipcode/atm/bank/Account.java

@@ -40,4 +40,9 @@ public abstract class Account {
         accountData = new AccountData(accountData.getId(), accountData.getName(), accountData.getEmail(),
                 newBalance);
     }
+
+
+    public void balance(int amount) {
+
+    }
 }

src/main/java/rocks/zipcode/atm/bank/Bank.java

@@ -49,4 +49,11 @@ public class Bank {
             return ActionResult.fail("Withdraw failed: " + amount + ". Account has: " + account.getBalance());
         }
     }
+
+    public <T> ActionResult<T> balance(AccountData accountData, int amount) {
+        Account account = accounts.get(accountData.getId());
+        account.balance(amount);
+
+        return (ActionResult<T>) ActionResult.success(account.getAccountData());
+    }
 }