begin function-python ex 2