Bug fix & GUI update