Slightly edited

When running the code, it seems like it was slightly obsolete - made a couple of changes.