assign4 WIP update