Improve styling on setup screen