add function
fixed issue