Add logical flag
Unit test for logical arg