Bentuk c++ menentukan bangun persegi atau persegi panjang

bentuk c++ menggunakan class :

#include <iostream>

istream& operator>>(istream& in, bangun& masukan){
         cout<<"Masukan a = ";
         in>>masukan.a;
         cout<<"Masukan b = ";
         in>>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;