added more assignment code