Added argument in the function