Add buttons and make visual changes