line problem py