final tweaks from referee comments