Add partials and update templateUrls to match