Added directory support for ui generator