minor tweaks
updated for paper submission