spark python script works