Revert "Remove unneeded button"