Cleanup; added basic DOM manipulation to directive