用matlab编写函数求矩形面积

怎样才能做到“用户不输入参数显示提示信息，用户输入一个参数不报错，而是直接当正方形算”？
求大佬解答解答吧，谢谢了。

用nargin参数

function area = rect(a,b)
if(nargin==0)
error('输入参数数目不足');
elseif(nargin==1)
area = a*a;
elseif(nargin==2)
area = a*b;
elseif(nargin>2)
error('太多输入参数了！')
end
end