UI & bugs fixed