Add query decorator