CT4003 /assignment two code