Fixed add function, no more Name Error.