Add optional directory output