Added class for custom drag function. Will test and use it now