add ui, fixes task bugs