Lab4 add error msg