Change declaration on add function