Updated step 4 test