Add public directory support