Fix display, add css