Fixed GUI to deal with new interface