Cleaned up the documentation on the code; Added important point"
added if