Spruced up comments in code