Add ui and change on some class