main() added in homework1.py