Added DataFrame support