Add spark sql support