Improve styling on the steps builder