Pisanje preproste funkcije (python)