Writing a simple function decorator