Replaced drag logic with brush logic