Python decorator example code