sorted out minor changes related to output -> two models dir change