Added unit tests for arguments