Feel free to change it. I won't have time yo merge it so feel free to steal the code :-)