Comments and small change. 

I commented a little so I know what I did, and there was a line that could be a little more specific and robust, so I made it so.