Added correct function arguments