Added directive to main module