corrections

correction on code