Add drag logic