Add parameter validation in main class