Add back toggle button translation transition