Working on fixing GUI errors