339 - tweaks to hw4