Add profile decorators to important functions