Cleanup: Allow @function decorator to decorate methods