Adding `removeColumn` function to visualization api mock