refactor: minor layout changes