if you like this changes, maybe you can merge it

Sorry this is my first time on github; I hope I have used the correct function.