improve: to check the arguments with switch statement.