fixed display to work much better