Adding required code for assignment