Added new assignment code