step 3.2 - minor bug fixed