Function now receive arguments