Add types for drag drata