step 5: added main