Fixed some visual problems