assign2 WIP