Support only one parameter

Making square in this case