another small tweak hw4b