fixed some visual bugs