Shortened some code for you

While peeking at your code, I came across some annoyingly verbose code; fixed that for ya. :D