Cleaned up code and corrected info section.
Cleaned up code. Used enumeration instead of integer in comparison.