Added import statement
Added auto directory creation