ui work and screenshot update