making corrections to the code