Improving performance.

Man, I was reviewing your code and decided to improve it a little bit :)