HW4: model changes, add to writeup.