Made the code a little clearer.