Add display function in helper