add ui picture and cleaned code