fix the second button

Please review my code