public void setDifficulty(Difficulty difficulty) {
        this.difficulty = difficulty;

        switch (difficulty.getDifficulty()) {
            case "medium":
                this.numberOfMines = 40;
                this.boardWidth = 16;
                this.boardHeight = boardWidth;
                break;
            case "hard":
                this.numberOfMines = 99;
                this.boardWidth = 16;
                this.boardHeight = 40;
                break;
            default:
                this.numberOfMines = 10;
                this.boardWidth = 8;
                this.boardHeight = boardWidth;
                break;
        }
        this.numberOfFlags = this.numberOfMines;
        this.board = new Cell[boardWidth][boardHeight];
    }