Corrected Code

Correctin some code.