GUI update - small fixes