Code tweaks per @metadave