minor tweaks forgot about last time