Test added for just the button