Now using correct function parameters