Updated setUpModule

setUpModule now includes TEMPLATE_DIRS as well