fixed visual bugs