correction of all code