rectified the code
rectified code