Add main function to HW5
Not sure how to test exercise 5