Разбить программу на функции

#include <iostream>
 
using namespace std;
 
int main()
{
    setlocale(LC_ALL, "Russian"); //Включаем русский язык
    int n;
    cout << "Введите размер массива: ";
    cin >> n;
    int ** A = new int * [n]; //Создаем динамический массив...
    for(int i = 0; i<n; i++)
    {
        A[i] = new int [n];
    }
    int *B = new int[n]; //...затем второй...
    cout <<"Заполите массив: \n\n" ; //...и заполняем его!^^
    for (int i=0; i<n; i++)
        {
            for (int j=0; j<n; j++)
            {
              cout << "Введите элемент [" << i + 1 << "][" << j + 1 << "]: ";
              cin >> A[i][j];
            }
    }
    for(int i = 0; i < n; i++) //считаем среднее арифметическое главной и побочной диагоналей
    {
        B[i] = (A[i][i]+A[i][n-1-i])/2; //а так же записываем его во второй динамический массив :З
    }
    cout << "Вы ввели массив: "; 
    for(int i = 0; i<n; i++)
    {
        cout << endl;
        for(int j = 0; j<n; j++)
        cout << A[i][j] << " "; //показываем введенный массив
    }
    cout << endl << "Полученный массив: ";
    for(int i = 0; i<n; i++)
    {
        cout << B[i] << " "; //показываем полученный массив
    }
    delete [] A; //очищаем память
    delete [] B;
    system("pause"); 
    return 0; 
}

#include "stdafx.h"
#include <iostream>
 
using namespace std;
 
void Input(int **A,int n)
{
    cout <<"Заполите массив: \n\n" ; //...и заполняем его!^^
    for (int i=0; i<n; i++)
        {
            for (int j=0; j<n; j++)
            {
              cout << "Введите элемент [" << i + 1 << "][" << j + 1 << "]: ";
              cin >> A[i][j];
            }
    }
}
 
void Display(int **A,int n)
{
    cout << "Вы ввели массив: "; 
    for(int i = 0; i<n; i++)
    {
        cout << endl;
        for(int j = 0; j<n; j++)
        cout << A[i][j] << " "; //показываем введенный массив
    }
 
 
}
 
void Obr(int **A,int *B,int n)
{
    for(int i = 0; i < n; i++) //считаем среднее арифметическое главной и побочной диагоналей
        B[i] = (A[i][i]+A[i][n-1-i])/2; //а так же записываем его во второй динамический массив :З
    for(int i = 0; i<n; i++)
        cout << B[i] << " "; //показываем полученный массив
}
 
int main()
{
    setlocale(LC_ALL, "Russian"); //Включаем русский язык
    int n;
    cout << "Введите размер массива: ";
    cin >> n;
    int ** A = new int * [n]; //Создаем динамический массив...
    for(int i = 0; i<n; i++)
        A[i] = new int [n];
 
    int *B = new int[n]; //...затем второй...
    Input(A,n);
    Display(A,n);
    cout << endl << "Полученный массив: ";
    Obr(A,B,n);
    delete [] A; //очищаем память
    delete [] B;
    system("pause"); 
    return 0; 
}