One-line program in Python