Added in missing parts of the directive