updated comment and changed the arg