Fix add function