couple small tweaks