one line code python