Fixed some visual problems in code