Python decorator fun!