improved part of ui