Updated to have the button as img