add back real time display, fix the previous bug