small cleanup/update