Update foundations_hw_4_2_census.py