Add partials and fix bug of custom directive not rendering