python home work