Added InputFormat and mapPartition