made a couple minor tweaks