program c++ yang menentukan bangun persegi atau persegi panjang

#include <iostream>

#include <string>

istream& operator>>(istream& in, bangun& masukan){

         cout<<"Masukan angka A = ";

         cin>>masukan.A;

         cout<<"Masukan anka B = ";

         cin>>masukan.B;  

         }

void bangun::menentukan(){

     if (A==B)

        cout<<"Bangun tersebut adalah Persegi"<<endl;

     else

        cout<<"Bangun tersebut adalah Persegi Panjang"<<endl;

        }

main()

{

   bangun X;

   cin>>X;

   X.menentukan();

  

   system ("Pause");

   return 0;

}

#include <string>

using namespace std;

class bangun {

      friend istream& operator>>(istream&, bangun&);

      public:

             void menentukan();

      private:

              int A,B;

              };

system ("pause");
return 0;