Improved comments in code