adding ability to pass entity name to directive