Add assignment code initial commit