Python home work