made a couple small tweaks