Fixed the looks a bit