Spark UDF in Scala and Python