Add functions to add/remove data