Add tags and fix bug