Add() function updated