fixed display bugs + display tests