Assignment Code Updated