add function bug fixed