Fix code style issues

I wrote this code around 3 AM last night so I missed a couple things that
I needed to do.