Fix tests
Add tag function