modified for Spark 2.0.0
update readme