
/**
 * Write a description of class SmallBank here.
 *
 * @author (your name)
 * @version (a version number or a date)
 */
public class MediumBank
{
    private BankAccount account1;
    private BankAccount account2;
    
    public MediumBank(String accountName, int balance)
    {
        account1 = new BankAccount (accountName,balance);
        account2 = null;
    }
    
    public void addSecondAccount(String accountName, int balance)
    {
        account2 = new BankAccount (accountName, balance);
    }
    
    public boolean deposit(String name,int amount) {
        if(account1.getName().equals(name)) {
            account1.addBalance(amount);
            return true;
        } else if(account2.getName().equals(name)) {
            account2.addBalance(amount);
            return true;
        } else {
            return false;
        }
    }
    
    public int getNumberOfAccounts() {
        if(account2!=null) {
            return 2;
        } else {
            return 1;
        }
    }
    public int totalBalance() {
        if(account2!=null) {
            return account1.getBalance() + account2.getBalance(); 
        } else {
            return account1.getBalance();
        }
    }
    
    public void printBankDetails()
    {
        if(account2 != null) {
           System.out.println(account1);
           System.out.println(account2);
        } else {
          System.out.println(account1);  
        }
    }
}
