Fixing so that citations are saved and removed properly