Simplified it and gave an id to the button itself