A few line of code in python