slight ui improvement - layout