Add code for assignment