Fix the "Add" function