Add optional validation