code corrections

corrected code for better readability;