Correcting some bugs

Correcting some visual bugs.