Add create_directory function