fixing little display bugs