hacked display logic a bit