Adding assignment code