Modified validation to include <button> element