Step 5: Added main