fix ui & bugs