Add "display on flat" feature