Memperbaiki issue #2 dan menambah UI