Improved comments slightly