Added code for assignment