Add support to specify root_dir