Add second button
tmp fix