tweaks removing unused import