Update comments with correct function params