Add one more condition in if.
Change argument order