step 5 complete. Add a main