Made some changes

And they were good