I made a few small changes