updated display.ml to account for the new types