fixing small visual bugs