My python lab homework