@Override
    public List<FriendlyPiece> getSetup() {
        List<FriendlyPiece> friendlyArmy = super.getSetup();
        ArrayList<HumanEnemyPiece>enemyArmy = new ArrayList<>();
        for(int i = 0; i < 4; i++) {
            for(int j=0; j<10; j++) {
                
                HumanEnemyPiece p = 
                        (color == Color.BLUE)? 
                            new HumanEnemyPiece(Color.RED, 9-i, 9-j):
                            new HumanEnemyPiece(Color.BLUE, i, j);
                enemyArmy.add(p);
            }
        }
        List<? extends Piece>redArmy;
        List<? extends Piece>blueArmy;
        if(color == Color.BLUE) {
            redArmy = enemyArmy;
            blueArmy = friendlyArmy;
        } else {
            redArmy = friendlyArmy;
            blueArmy = enemyArmy;
        }
        state = new GameState(Color.BLUE, blueArmy, redArmy);
        startState = state;
        return friendlyArmy;
    }