Fix tagbox & tidy up