Enhanced argument validation in proof generator