Moved drop behavior into widget