Remove extra lines in the code