Fix missing div, minor cosmetic changes