Step 5 implemented
Introduced support for If and While statements