Small design update