moved remove() to directive