Corrected an obvious mistake and added a comment.
Cleanup Code

Cleaned up some messy code.