Added comments to if / else statement