change n, in main, to 3