Fixed visual bugs