I made some little changes

I made some little changes in the few minutes I had to play with it