adding assignment code