Missing semicolon
Better button state logic