UI change finished stage 2