Made the code a little cleaner.