homework task on python