Added  injection to directive