Add remove by id function