Fix functionality to edit tags but may need to revisit

For some reason I need to iterate over the tags even if I technically have one in there, will need to take a look later but for now it works