add some more logic to ui module and add initial stylesheet