hw5 tweaks & fixes