Add code pass .ex4 test