Corrections

Code corrections