Added expect code