step 1.4 - identation bug fixed