minor changes and ui