UI update. UI now correctly positioned