added logic around button display. new tests to be solved for.