Modifications on the code

Had some extra lines that were not deleted properly.