A few minor adjustments

Fixed/added comments. Removed extra lines.