add argument for function