made a couple tweaks