add code to button
add spacing under image