Add correction GUI functionality