Improve benefits layout for all screens