added code

Hello Louis

I changed your code.

What do you think?

-Vaed