Cleaned

Cleaned the code