fixing visual bugs