WIP - Test added for just the button