Updated the code, repaired something