Touched up the code a little bit, saving some bytes.