Add create_dir function and fix test