Code cleanup

Added simple comments for readability.