Fix - GUI bugs