Managed to get add and remove working with the database (we think).