Finish restock on the UI with flash implementation