Cleaned the Code