add some assignment code, tests needed