Small improvement
Added option to use hdfs