cleaned the code for submision